How to change color of the graph in python